package com.designPattern.flyweight;

import java.util.ArrayList;
import java.util.List;

/**
 * 模拟数据库连接池
 */
public class ConnectionPool {
    //用于存放连接对象的链表
    private List<Connection> connectionPool = new ArrayList<>();
    //默认的数据库连接池中的连接对象
    private final int DEFAULT_CONNECTION = 5;
    //数据库中的连接对象数量
    private int connectionNum = DEFAULT_CONNECTION;
    //【不能用】设置连接池数量（这里暂时没有什么用，也无法实现对数据库连接池数量的控制）
    public void setConnectionNum(int connectionNum) {
        this.connectionNum = connectionNum;
    }
    //获取连接池数量
    public int getConnectionNum() {
        return connectionNum;
    }
    public ConnectionPool(){
        for (int i = 0; i < connectionNum; i++) {
            connectionPool.add(new Connection(false,null));
        }
    }
    public ConnectionPool(int connectionNum) throws Exception {
        if (connectionNum>0) {
            for (int i = 0; i < connectionNum; i++) {
                connectionPool.add(new Connection(false,null));
            }
        }else {
            //lower than zero
            throw new Exception("the connectionNum is lower than 0!");
        }
    }

    //增加数据库连接池
    public void addConnection(){
        connectionPool.add(new Connection(false,null));
        this.connectionNum++;
    }
    //判断数据库连接池是否为空
    public boolean isEmpty(){
        return connectionNum == 0;
    }
    //获取数据库连接
    public Connection getConnection(){
       if (!isEmpty()){
           //连接池数量 - 1
           connectionNum--;
           Connection connection = connectionPool.get(0);
           //将连接状态更改
           connection.setOnLine(true);
           connectionPool.remove(0);
           return connection;
       }
       //如果池中已经没有连接则新增数据库连接
        addConnection();
        return getConnection();
    }
    //关闭数据库中的某个连接
    public void closeConnection(Connection connection){
        connection.setOnLine(false);
        //将这个连接对象放回到池中
        connectionPool.add(connection);
        connectionNum++;
    }
}
